# 自己封装的接口
import requests
import app


# 封装 login 类，添加登录使用的方法
class LoginAPI:
    def __init__(self):
        self.url = app.BASE_URL + '/api/sys/login'
        self.headers = app.HEADERS

    def login(self, json_data):
        return requests.post(url=self.url, headers=self.headers, json=json_data)


class EmpManageApi:

    def __init__(self):
        self.url = app.BASE_URL + '/api/sys/user'

    # 添加员工
    def add_emp(self, header, json_data):
        resp = requests.post(url=self.url, headers=header, json=json_data)
        # 获取员ID并写入全局变量
        app.EMP_ID = resp.json().get('data').get('id')
        return resp

    # 查询员工
    def query_emp(self, header):
        resp = requests.get(url=self.url + '/' + app.EMP_ID, headers=header)
        return resp

    # 修改员工
    def modify_emp(self, header, json_data):
        resp = requests.put(url=self.url + '/' + app.EMP_ID,
                            headers=header, json=json_data)
        return resp

    # 删除员工
    def delete_emp(self, header):
        resp = requests.delete(url=self.url + '/' + app.EMP_ID, headers=header)
        return resp

    # 查询员列表
    def query_emp_list(self, header):
        resp = requests.get(url=self.url + '?' + 'page=1&size=10',
                            headers=header)
        return resp
